package com.ita.acmsystem.merchant.servlet;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.List;
import java.util.Properties;
import java.util.UUID;

import javax.jms.JMSException;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import com.ita.acmsystem.merchant.config.Constant;
import com.ita.acmsystem.merchant.entity.Merchant;
import com.ita.acmsystem.merchant.service.IMerchantService;
import com.ita.acmsystem.merchant.service.impl.MerchantServiceImpl;
import com.ita.acmsystem.merchant.util.JMSUtil;
import com.ita.acmsystem.merchant.util.JsonUtil;

/**
 * Servlet implementation class RegisterServlet
 */
public class RegisterServlet extends HttpServlet {
	private static final long serialVersionUID = 1L;

	private IMerchantService merchantService = new MerchantServiceImpl();

	// 上传文件存储目录
	private static final String MERCHANT_PICTURE_DIRECTORY = "merchant";
	private static final String IDCARD_PICTURE_DIRECTORY = "idcard";

	// 上传配置
	private static final int MEMORY_THRESHOLD = 1024 * 1024 * 3; // 3MB
	private static final int MAX_FILE_SIZE = 1024 * 1024 * 40; // 40MB
	private static final int MAX_REQUEST_SIZE = 1024 * 1024 * 50; // 50MB

	/**
	 * @see HttpServlet#HttpServlet()
	 */
	public RegisterServlet() {
		super();
		// TODO Auto-generated constructor stub
	}

	/**
	 * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse
	 *      response)
	 */
	protected void doGet(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		// TODO Auto-generated method stub
	}

	/**
	 * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse
	 *      response)
	 */
	protected void doPost(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		// TODO Auto-generated method stub
		doRegister(request, response);
	}

	private void doRegister(HttpServletRequest request,
			HttpServletResponse response) throws IOException {
		// 检测是否为多媒体上传
		if (!ServletFileUpload.isMultipartContent(request)) {
			// 如果不是则停止
			PrintWriter writer = null;
			try {
				writer = response.getWriter();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
			writer.println("Error: 表单必须包含 enctype=multipart/form-data");
			writer.flush();
			return;
		}

		// 配置上传参数
		DiskFileItemFactory factory = new DiskFileItemFactory();
		// 设置内存临界值 - 超过后将产生临时文件并存储于临时目录中
		factory.setSizeThreshold(MEMORY_THRESHOLD);
		// 设置临时存储目录
		factory.setRepository(new File(System.getProperty("java.io.tmpdir")));

		ServletFileUpload upload = new ServletFileUpload(factory);
		// 设置最大文件上传值
		upload.setFileSizeMax(MAX_FILE_SIZE);
		// 设置最大请求值 (包含文件和表单数据)
		upload.setSizeMax(MAX_REQUEST_SIZE);
		// 中文处理
		upload.setHeaderEncoding("UTF-8");

		// 构造临时路径来存储上传的文件
		// 这个路径相对当前应用的目录
		String merchantPath = getServletContext().getRealPath("/")
				+ File.separator + MERCHANT_PICTURE_DIRECTORY;
		String idcardPath = getServletContext().getRealPath("/")
				+ File.separator + IDCARD_PICTURE_DIRECTORY;
		// 如果目录不存在则创建
		File merchantDir = new File(merchantPath);
		if (!merchantDir.exists()) {
			merchantDir.mkdir();
		}
		File idcardDir = new File(idcardPath);
		if (!idcardDir.exists()) {
			idcardDir.mkdir();
		}

		// 获取表单中的文本字段
		String id = UUID.randomUUID().toString();
		HttpSession session = request.getSession();
		Merchant merchant = (Merchant) session.getAttribute("user");
		if (merchant != null) {
			id = merchant.getMerchantId();
		}
		String name = null;
		String address = null;
		String password = null;
		File merchantFile = null;
		File idcardFile = null;

		String marchentPicSavePath = null;
		String idcardPicSavePath = null;

		// 解析请求的内容提取文件数据
		List<FileItem> formItems = null;
		try {
			formItems = upload.parseRequest(request);
		} catch (FileUploadException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		if (formItems != null && formItems.size() > 0) {
			// 迭代表单数据
			for (FileItem item : formItems) {
				// 处理不在表单中的字段
				if (!item.isFormField()) {
					if ("headPic".equals(item.getFieldName())) {

						String fileName = id;
						String filePath = merchantPath + File.separator + id;

						// 从配置文件中获得host和port
						InputStream is = RegisterServlet.class.getClassLoader()
								.getResourceAsStream("property.propertties");
						Properties prop = new Properties();
						prop.load(is);
						String host = prop.getProperty("host");
						String port = prop.getProperty("port");

						// 入库地址
						marchentPicSavePath = "http://" + host + ":" + port
								+ "/./merchant/" + id;

						File storeFile = new File(filePath);
						// 在控制台输出文件的上传路径
						System.out.println(filePath);
						// 保存文件到硬盘
						try {
							item.write(storeFile);
						} catch (Exception e) {
							// TODO Auto-generated catch block
							e.printStackTrace();
						}
						merchantFile = storeFile;
						request.setAttribute("message", "文件上传成功，注册审核中");
					} else if ("idCard".equals(item.getFieldName())) {

						String fileName = id;
						String filePath = idcardPath + File.separator
								+ fileName;
						
						// 从配置文件中获得host和port
						InputStream is = RegisterServlet.class.getClassLoader()
								.getResourceAsStream("property.propertties");
						Properties prop = new Properties();
						prop.load(is);
						String host = prop.getProperty("host");
						String port = prop.getProperty("port");

						// 入库地址
						idcardPicSavePath = "http://" + host + ":" + port
								+ "/./idcard/" + id;
						
						File storeFile = new File(filePath);
						// 在控制台输出文件的上传路径
						System.out.println(filePath);
						// 保存文件到硬盘
						try {
							item.write(storeFile);
						} catch (Exception e) {
							// TODO Auto-generated catch block
							e.printStackTrace();
						}
						idcardFile = storeFile;
						request.setAttribute("message", "文件上传成功,注册审核中");
					}
				} else {
					switch (item.getFieldName()) {
					case "name":
						name = item.getString("UTF-8");
						break;
					case "address":
						address = item.getString("UTF-8");
						break;
					case "password":
						password = item.getString();
						break;
					}
				}
			}
		}

		Merchant registerUser = new Merchant(id, name, password, address,
				merchantFile.getPath(), idcardFile.getPath(),
				Constant.WHITE_LIST);
		Merchant registerUserLocal = new Merchant(id, name, password, address,
				marchentPicSavePath, idcardPicSavePath, Constant.WHITE_LIST);
		// merchantService.addMerchant(registerUser);
		merchantService.addMerchant(registerUserLocal);
		System.out.println(registerUser);

		String merchantJsonStr = JsonUtil
				.generateJsonFromMerchant(registerUser);
		boolean sendToMQ = false;
		try {
			sendToMQ = JMSUtil.sendMessageToQueue(merchantJsonStr);
		} catch (JMSException e1) {
			// TODO Auto-generated catch block
			e1.printStackTrace();
		}
		if (!sendToMQ) {
			request.setAttribute("message", "注册失败");
		}

		try {
			getServletContext().getRequestDispatcher("/afterRegister.jsp")
					.forward(request, response);
		} catch (ServletException | IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
}
